package com.zhaoqi.hr.repository;

import com.zhaoqi.hr.entity.User;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.JpaSpecificationExecutor;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Optional;

/**
 * 用户数据访问层
 */
@Repository
public interface UserRepository extends JpaRepository<User, Long>, JpaSpecificationExecutor<User> {

    /**
     * 根据用户名查找用户
     */
    Optional<User> findByUsername(String username);

    /**
     * 检查用户名是否存在
     */
    boolean existsByUsername(String username);

    /**
     * 根据手机号查找用户
     */
    Optional<User> findByPhone(String phone);

    /**
     * 检查手机号是否存在
     */
    boolean existsByPhone(String phone);

    /**
     * 检查身份证号是否存在
     */
    boolean existsByIdCard(String idCard);

    /**
     * 根据角色查找用户
     */
    List<User> findByRole(User.UserRole role);

    /**
     * 根据状态查找用户
     */
    List<User> findByStatus(User.UserStatus status);

    /**
     * 根据角色和状态查找用户
     */
    List<User> findByRoleAndStatus(User.UserRole role, User.UserStatus status);

    /**
     * 统计指定时间段内新注册用户数
     */
    @Query("SELECT COUNT(u) FROM User u WHERE u.createdAt BETWEEN :startTime AND :endTime")
    Long countNewUsersByPeriod(@Param("startTime") LocalDateTime startTime, 
                               @Param("endTime") LocalDateTime endTime);

    /**
     * 统计各角色用户数量
     */
    @Query("SELECT u.role, COUNT(u) FROM User u GROUP BY u.role")
    List<Object[]> countUsersByRole();

    /**
     * 根据角色统计用户数量
     */
    long countByRole(User.UserRole role);

    /**
     * 根据状态统计用户数量
     */
    long countByStatus(User.UserStatus status);

    /**
     * 根据姓名或手机号搜索用户（分页）
     */
    Page<User> findByRealNameContainingOrPhoneContaining(String realName, String phone, Pageable pageable);
    
    /**
     * 根据用户名或真实姓名搜索用户（按创建时间倒序，分页）
     */
    List<User> findByUsernameContainingOrRealNameContainingOrderByCreatedAtDesc(String username, String realName, Pageable pageable);
    
    /**
     * 获取所有用户（按创建时间倒序，分页）
     */
    List<User> findAllByOrderByCreatedAtDesc(Pageable pageable);
}
